In [1]:
%run train-cGAN
# %run unit_test
self.device is cuda:0
64
Epoch: 0, Itration: 0
loss_G_GAN = 4.886846542358398, loss_G_L1 = 0.39250922203063965 * 5
loss_D = 0.7143234014511108
Epoch: 76, Itration: 1000
loss_G_GAN = 9.659367561340332, loss_G_L1 = 0.15902411937713623 * 5
loss_D = 0.006910865660756826
Epoch: 153, Itration: 2000
loss_G_GAN = 8.78011417388916, loss_G_L1 = 0.17440780997276306 * 5
loss_D = 0.0005042690900154412
save model to ./checkpoints/pix2food.pkl at iteration = 3000
Epoch: 230, Itration: 3000
loss_G_GAN = 4.836752891540527, loss_G_L1 = 0.17835037410259247 * 5
loss_D = 0.18689367175102234
Finally save model to ./checkpoints/pix2food.pkl



 # --- Validation Set --- # 



In [3]:
# %run train-cGAN
In [2]:
# %run train
In [ ]:
 
In [ ]: